"""
示例：
请选择需要执行的操作：
1. WAV 转 MP3
2. MP3 转 WAV
3. 退出程序
请输入数字选择操作：2
请输入需要转换的 MP3 文件名：p:\a\1.mp3
请输入要保存的 WAV 文件名：p:\b\1.wav
转换完成！
"""

from pydub import AudioSegment
import os

print("欢迎使用音频文件转换器")

while True:
    # 打印菜单
    print("\n请选择需要执行的操作：")
    print("1. WAV 转 MP3")
    print("2. MP3 转 WAV")
    print("3. 退出程序")

    # 获取用户输入
    choice = input("请输入数字选择操作：")

    # 执行相应操作
    if choice == "1":
        # WAV 转 MP3
        wav_file = input("请输入需要转换的 WAV 文件名：")
        if not os.path.isfile(wav_file):
            print("错误：文件不存在")
            continue
        mp3_file = input("请输入要保存的 MP3 文件名：")
        AudioSegment.from_wav(wav_file).export(mp3_file, format="mp3")
        print("转换完成！")
    elif choice == "2":
        # MP3 转 WAV
        mp3_file = input("请输入需要转换的 MP3 文件名：")
        if not os.path.isfile(mp3_file):
            print("错误：文件不存在")
            continue
        wav_file = input("请输入要保存的 WAV 文件名：")
        AudioSegment.from_mp3(mp3_file).export(wav_file, format="wav")
        print("转换完成！")
    elif choice == "3":
        # 退出程序
        print("谢谢使用，再见！")
        break
    else:
        print("错误：无效的选项")
